from factory import DjangoModelFactory, Faker

from .models import Movie


class MovieFactory(DjangoModelFactory):
    """电影工厂类，用于创建测试用的电影实例"""

    class Meta:
        model = Movie  # 关联的模型类

    title = Faker("sentence", nb_words=4)  # 电影标题
    genres = Faker(
        "pylist", nb_elements=3, variable_nb_elements=True, value_types=["str"]
    )  # 电影类型列表
